package com.example.llm.advisor;

import com.example.llm.OllamaClient.Advisor;
import com.google.gson.JsonObject;

public class QuestionAnswerAdvisor implements Advisor {
    private final String query;
    private final int retrieveSize;

    public QuestionAnswerAdvisor(String query, int retrieveSize) {
        this.query = query;
        this.retrieveSize = retrieveSize;
    }

    @Override
    public void apply(JsonObject json) {
        JsonObject context = new JsonObject();
        context.addProperty("query", query);
        context.addProperty("retrieve_size", retrieveSize);
        json.add("context", context);
    }
} 